package com.xlh.openai.service;

import com.baomidou.mybatisplus.extension.service.IService;
import com.plexpt.chatgpt.entity.chat.Message;
import com.xlh.openai.entity.ChatRecordsEntity;

import java.util.List;

/**
 * 
 *
 * @author 1280251739@qq.com
 * @email sunlightcs@gmail.com
 * @date 2023-04-07 09:10:09
 */
public interface ChatRecordsService extends IService<ChatRecordsEntity> {

    /**
     * 保存对话
     * @param id
     * @param question
     * @param answers
     */
    void saveConversation(Integer id, Message question, Message answers);

    /**
     * 获取用户对话列表
     * @param userId
     * @return
     */
    List<Message> listMessages(Integer userId);

    /**
     * 根据用户ID和会话ID查询信息
     * @param userId
     * @param conversationId
     * @return
     */
    List<Message> listMessages(Integer userId, String conversationId);

    /**
     * 根据用户ID和会话ID查询信息
     * @param userId
     * @param conversationId
     * @return
     */
    List<Message> listMessagesTest(Integer userId, String conversationId);


    /**
     * 保存用户信息
     * @param userId 用户ID
     * @param conversationId 会话ID
     * @param prompt 提示语
     * @param answer 回答
     */
    void saveConversation(Integer userId, String conversationId, String prompt, String answer, Long tokens);



}

